﻿#region GNU General Public License
/*
-----------------------------------------------------------------------------
This source file is part of jscc
    (Javascript with Classes)
For the latest info, see http://code.google.com/p/jscc

Copyright (c) 2011 Tapani Valkonen

jscc is free software: you can redistribute it and/or modify it under 
the terms of the GNU General Public License as published by the 
Free Software Foundation, either version 3 of the License, or (at your option)
any later version. 
 
jscc is distributed in the hope that it will be useful, 
but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 
for more details.

You should have received a copy of the GNU General Public License 
along with jscc. If not, see http://www.gnu.org/licenses/.
-----------------------------------------------------------------------------
*/
#endregion

using System;
using System.Collections.Generic;
using System.Text;
using jscc.Ast;
using System.IO;

namespace jscc
{
    public class GeneratorVisitor : AstNodeVisitor
    {
        CodeWriter writer;

        public GeneratorVisitor(CodeWriter writer)
        {
            this.writer = writer;
        }

        public override void Visit(AstNode node)
        {
            if(node.Writer != null)
                node.Writer.Write(writer);
        }

        public override void Leave(AstNode node)
        {
            if (node.Writer != null)
                node.Writer.WriteEnd(writer);
        }
    }

    public class ArrangeVisitor : AstNodeVisitor
    {
        Dictionary<AstClass, AstConstructor> constructors = new Dictionary<AstClass, AstConstructor>();

        public ArrangeVisitor()
        {
        }

        public void Arrange()
        {
            foreach (KeyValuePair<AstClass, AstConstructor> entry in constructors)
            {
                entry.Key.MakeLastChildren(entry.Value);
            }
        }

        public override void Visit(AstNode node)
        {
            AstClass klass = node.Parent as AstClass;
            AstConstructor ctr = node as AstConstructor;

            if (node is AstConstructor && klass != null)
            {
                constructors[klass] = ctr;
            }
        }
    }

    public class Generator
    {
        List<File> files;

        public Generator()
        {
            files = new List<File>();
        }

        public void Generate(AstNode root, string filename)
        {
            ArrangeVisitor arrange = new ArrangeVisitor();
            root.Accept(arrange);
            arrange.Arrange();

            CodeWriter writer = new JsCodeWriter();
            GeneratorVisitor visitor = new GeneratorVisitor(writer);
            root.Accept(visitor);

            files.Add(new File(filename, writer.Content));
        }

        public void PrintFiles()
        {
            foreach(File file in files)
            {
                Console.WriteLine(file.FileName);
                Console.WriteLine(file.Content);
            }
        }

        public void WriteFiles()
        {
            foreach (File file in files)
            {
                StreamWriter stream = System.IO.File.AppendText(file.FileName);
                stream.Write(file.Content);
                stream.Close();
            }
        }
    }
}
